import com.fasterxml.jackson.databind.ObjectMapper;
import com.mysql.cj.jdbc.MysqlDataSource;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.sql.DataSource;
import java.io.IOException;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

class Message {
    public String from;
    public String to;
    public String message;

    @Override
    public String toString() {
        return "Message{" +
                "from='" + from + '\'' +
                ", to='" + to + '\'' +
                ", message='" + message + '\'' +
                '}';
    }
}

@WebServlet("/message")
public class MessageServlet extends HttpServlet {

    // 使用数据库存储数据  不再需要 messageList
//    private List<Message> messageList = new ArrayList<>();
    private ObjectMapper objectMapper = new ObjectMapper();

    // 创建数据源
    private DataSource dataSource = new MysqlDataSource();

    @Override
    public void init() throws ServletException {
        // 连接数据库
        ((MysqlDataSource)dataSource).setURL("jdbc:mysql://127.0.0.1:3306/message_wall?characterEncoding=utf8&useSSL=false");
        ((MysqlDataSource)dataSource).setUser("root");
        ((MysqlDataSource)dataSource).setPassword("123456");
    }

    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        // 先获取前端发来的 json 字符串, 将 json 转换成 java 对象
        // 从前端获取到 json 字符串，这里通过读取流对象里的内容获取
        // 然后将读取到的 json 字符串，按照 json 格式，解析成 Map （键值对）
        // 然后根据第二参数，将 Map（键值对） 装换成 Java 对象
        Message message = objectMapper.readValue(req.getInputStream(), Message.class);
        System.out.println("请求中收到的Message：" + message);

        // 使用数据库存储数据  就不再需要 messageList 了
        // 将装换好的 message对象 添加到集合中
//        messageList.add(message);

        // 将转换好的 message 对象，添加到数据库中
        try {
            save(message);
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }

        // 设置状态码
        resp.setStatus(200);
        // 返回到浏览器中 要显示在浏览器中的内容
        resp.getWriter().write("ok");
//        resp.setContentType("application/json");
//        resp.getWriter().write("{ ok: true }");
    }

    private void save(Message message) throws SQLException {
        // 1. 建立连接
        Connection connection = dataSource.getConnection();

        // 2. 编写sql
        String sql = "insert into message values(?, ?, ?)";
        PreparedStatement statement = connection.prepareStatement(sql);
        statement.setString(1, message.from);
        statement.setString(2, message.to);
        statement.setString(3, message.message);

        // 3. 执行sql
        statement.executeUpdate();

        // 4. 释放资源
        statement.close();
        connection.close();
    }

    private List<Message> load() throws SQLException {
        List<Message> messageList = new ArrayList<>();

        // 1. 建立连接
        Connection connection = dataSource.getConnection();

        // 2. 编写sql
        String sql = "select * from message";
        PreparedStatement statement = connection.prepareStatement(sql);

        // 3. 获取结果集
        ResultSet resultSet = statement.executeQuery();

        // 4. 取出结果集中的每一条数据 添加到 messageList 中
        while (resultSet.next()) {
            Message message = new Message();
            message.from = resultSet.getString("from");
            message.to = resultSet.getString("to");
            message.message = resultSet.getString("message");
            messageList.add(message);
        }

        // 5. 释放资源
        resultSet.close();
        statement.close();
        connection.close();

        // 6. 返回 messageList
        return messageList;
    }

    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        // 通过这个方法来处理当前获取消息列表的 get 请求，不需要解析参数，直接返回数据即可
        // 把 messageList 中的内容返回给 客户端
        resp.setStatus(200);
        resp.setContentType("application/json; charset=utf8");

        // 使用数据库存储数据  不再需要 messageList 了
//        String respJson = objectMapper.writeValueAsString(messageList);

        // 从数据库中获取数据
        List<Message> messageList = null;
        try {
           messageList = load();
        } catch (SQLException e) {
            throw new RuntimeException(e);
        }

        // 将集合 转换成 json 字符串
        String respJson = objectMapper.writeValueAsString(messageList);

        resp.getWriter().write(respJson);
    }
}
